package cn.su.upload_download.utils;



import org.apache.tools.ant.BuildException;
import org.apache.tools.ant.Project;
import org.apache.tools.ant.taskdefs.Expand;
import org.apache.tools.ant.taskdefs.Zip;
import org.apache.tools.ant.types.FileSet;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;

import java.io.*;
import java.util.Enumeration;

public class CompressFileUtils {
    public final static String encoding = "GBK";

    /**
     * 压缩文件和文件夹
     *
     * @param srcPathname 需要被压缩的文件或文件夹路径
     * @param zipFilepath 将要生成的zip文件路径
     * @throws BuildException
     * @throws RuntimeException
     */
    public static void zip(String srcPathname, String zipFilepath) throws BuildException, RuntimeException {
        File file = new File(srcPathname);
        if (!file.exists()) {
            throw new RuntimeException("source file or directory " + srcPathname + " does not exist.");
        }

        Project proj = new Project();
        FileSet fileSet = new FileSet();
        fileSet.setProject(proj);
        // 判断是目录还是文件
        if (file.isDirectory()) {
            fileSet.setDir(file);
            // ant中include/exclude规则在此都可以使用
            // 比如:
            // fileSet.setExcludes("**/*.txt");
            // fileSet.setIncludes("**/*.xls");
        } else {
            fileSet.setFile(file);
        }

        Zip zip = new Zip();
        zip.setProject(proj);
        zip.setDestFile(new File(zipFilepath));
        zip.addFileset(fileSet);
        zip.setEncoding(encoding);
        zip.execute();

        System.out.println("compress successed.");
    }

    /**
     * 解压缩文件和文件夹
     *
     * @param zipFilepath 需要被解压的zip文件路径
     * @param destDir 将要被解压到哪个文件夹
     * @throws BuildException
     * @throws RuntimeException
     */
    public static void unzip(String zipFilepath, String destDir) throws BuildException, RuntimeException {
        if (!new File(zipFilepath).exists()) {
            throw new RuntimeException("zip file " + zipFilepath + " does not exist.");
        }

        Project proj = new Project();
        Expand expand = new Expand();
        expand.setProject(proj);
        expand.setTaskType("unzip");
        expand.setTaskName("unzip");
        expand.setEncoding(encoding);

        expand.setSrc(new File(zipFilepath));
        expand.setDest(new File(destDir));
        expand.execute();

        System.out.println("uncompress successed.");
    }


    public static void unZipFiles(String zipPath,String descDir) throws Exception {
        unZipFiles(new File(zipPath),descDir);
    }

    /**
     * 解压文件到指定目录
     * @param zipFile 压缩文件
     * @param descDir 指定目录
     */
    @SuppressWarnings("rawtypes")
    public static void unZipFiles(File zipFile,String descDir) throws IOException {
//		request.setCharacterEncoding("utf-8");
        File pathFile = new File(descDir);
        if (!pathFile.exists()){
            pathFile.mkdirs();
        }
        /**
         * ZipFile类用于从zip文件中读取条目
         * getEntries()返回ZIP文件条目中的枚举
         */
        ZipFile zip = new ZipFile(zipFile);
        for (Enumeration entries = zip.getEntries(); entries.hasMoreElements();){
            ZipEntry entry = (ZipEntry) entries.nextElement();
            entry.setUnixMode(644);//解决Linux乱码
            String zipEntryName = entry.getName();
            System.out.println(zipEntryName);
            InputStream in = zip.getInputStream(entry);
            BufferedInputStream bis = new BufferedInputStream(in);
            String outPath = (descDir+zipEntryName).replaceAll("\\*","/");
            //判断路径是否存在，不存在则创建文件路径
            File file = new File(outPath.substring(0,outPath.lastIndexOf("/")));
            if (!file.exists()){
                file.mkdirs();
            }
            //判断文件全路径是否为文件夹，如果是上面已经上传，不需要解压
            if (new File(outPath).isDirectory()){
                continue;
            }
            //输出文件路径信息
//			System.out.println(outPath);
            OutputStream out = new FileOutputStream(outPath);
            byte[] buf1 = new byte[1024];
            int len;
            while ((len = in.read(buf1)) > 0){
                out.write(buf1,0,len);
            }
            out.close();
            bis.close();
            in.close();
        }
        System.out.println("*******************解压完毕********************");
    }

    public static void main(String[] args) {
        try {
            unzip("C:\\Users\\11581\\Desktop/WEBGL.zip","E:/uploadDir/test/");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}